from torch import Tensor
from torch_geometric.nn import MessagePassing
from torch_geometric.typing import Adj


class MyMessagePassingModule(MessagePassing):
    def __init__(self):
        super().__init__()

    def forward(self, x, edge_index):
        norm = [1]

        out = self.propagate(edge_index=edge_index,x=x, norm=norm)  # 这里调用message函数，进行消息计算

        return out

    def message(self, x, norm):
        """
        计算每一次需要传递的消息
        """
        pass
